custom_imports = dict(
    imports=['custom.models.heads.transformer_roi_extractor', 'custom.ops.roi_attention'],
    allow_failed_imports=False)
_base_ = '../../configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py'
# model settings
model = dict(
    roi_head=dict(
        bbox_roi_extractor=dict(
            type='TransformerRoiExtractor',
            roi_layer=dict(type='RoIAlign', output_size=7, sampling_ratio=2),
            out_channels=256,
            featmap_strides=[4, 8, 16, 32],
            query_mode="max",
            attention_config=dict(
                 type="RoiAttention",
                 dim=256, 
                 heads = 8,
                 dim_head = 32,
                 dropout=0.)
    )))
